import numpy as np
import matplotlib.pyplot as plt

# 初始化一个6x6的二维数组，用于存储每个点数组合出现的次数
counts = np.zeros((6, 6))

# 模拟掷骰子10000次
for _ in range(10000):
    # 随机生成三个骰子的点数
    dice1 = np.random.randint(1, 7)
    dice2 = np.random.randint(1, 7)
    dice3 = np.random.randint(1, 7)
    
    # 增加对应点数组合的计数
    counts[dice1-1, dice2-1] += 1

# 创建热力图
plt.imshow(counts, cmap='hot', interpolation='nearest')
plt.colorbar(label='Frequency')

# 设置x轴和y轴的标签
plt.xlabel('Dice 1')
plt.ylabel('Dice 2')

# 设置x轴和y轴的刻度标签
plt.xticks(np.arange(6) + 0.5, range(1, 7))
plt.yticks(np.arange(6) + 0.5, range(1, 7))

# 设置标题
plt.title('Results of Rolling Three D6')

# 显示图形
plt.show()